"""
类名：LoginApi
实例属性：
login_url：保存登陆的url，"http://ihrm-test.itheima.net/api/sys/login"
实例方法：
    def login(self, body)：
        # 功能：发送登陆请求
        # :param body: 发送请求的请求体
        # :return: 发送登陆请求后的响应对象
"""
import requests
from config import URL
import logging


class LoginApi(object):
    def __init__(self):
        self.login_url = URL + "/api/sys/login"

    def login(self, body):
        """
        # 功能：发送登陆请求
        # :param body: 发送请求的请求体
        # :return: 发送登陆请求后的响应对象
        """
        resp = requests.post(self.login_url, json=body)
        return resp

    def get_au_headers(self, body):
        """
        功能：
        :param body:
        :return:请求头
        """
        # 1. 发送登录请求
        resp = self.login(body)
        # 2. 获取响应内容的json格式
        json_data = resp.json()
        print(f'get_au_headers的json_data：{json_data}')
        logging.info(f'get_au_headers的json_data：{json_data}')
        # 3. 返回请求头
        return {
            "Authorization": json_data['data']
        }
